#ifndef KPEXTOPIMGF32C1MULPOW_H
#define KPEXTOPIMGF32C1MULPOW_H

#include "kpextVkContext.h"
#include "kpextBaseCompute.h"

namespace kpext {

/**
 * @brief 单通道32位单精度浮点数图像乘以a, 阈值后进行幂运算，规范化到0~65535， 一般用于调整亮度和对比度
 */
class OpImgF32C1MulPow : public BaseCompute{
public:
    explicit OpImgF32C1MulPow(kpextVkContext * ctx);

    virtual ~OpImgF32C1MulPow() override;

    /**
     * @brief  重新构建计算管线及相关资源
     * @param cols  输入输出二维图像的列数
     * @param rows  输入输出二维图像的行数
     * @param thres 阈值
     * @param b  阈值前乘以a值加上的偏移值
     * @param a  阈值前乘以的值
     * @param y  阈值后幂运算次方值
     * @param inbuf  输入缓冲
     * @param outbuf  输出缓冲
     */
    virtual void rebuild(int cols,
                         int rows,
                         uint32_t thres,
                         float b,
                         float a,
                         float y,
                         std::shared_ptr<BaseBuffer> inbuf,
                         std::shared_ptr<BaseBuffer> outbuf
                         );

protected:

    /**
     * @brief 计算着色器的work group size
     */
    std::array<uint32_t, 3> m_workGroupSize;

private:
    using BaseCompute::rebuild;

};  // class kpext::OpImgF32C1MulPow

} // namespace kpext

#endif // KPEXTOPIMGF32C1MULPOW_H
